[AI人工智能] 二、安裝python虛擬環境 + pytorch


Posted by mike-hsieh on 2024-01-13

紀錄 python安裝虛擬環境的過程。
切記,版本一定要照 [AI人工智能] 一、Python環境安裝 這裡的步驟安裝,否則跑程式很有可能會出現未知錯誤喔!

由於學習跟專案會拆兩邊,所以我會建議安裝兩個環境,一個是 Anaconda 的,另一個是 Pycharm 的,步驟基本如下,就看各位要建幾個。

  1. 新建一個 python 虛擬環境:
-- 新建一個名為 python38_cuda_conda 的 python環境,版本為3.8
conda create -n python38_cuda_conda python=3.8

-- 切換到 python38_cuda_conda
-- 補充一下: 使用 activate 後,後續的安裝都建議使用 pip 來安裝,pip 提供比較齊全。
conda activate python38_cuda_conda

-- 安裝 torch 可以參考以下路徑:
-- 1. https://pytorch.org/get-started/locally/
-- 2. https://pytorch.org/get-started/previous-versions/
-- 3. torch有分 CPU, GPU 兩種版本,如果有顯卡就可以使用GPU版本。

-- GPU版本(需先查詢是否有顯卡,以及對應的cuda版本)
cmd -> nvidia-smi
-- 安裝指令類似如下,詳細就可以參考
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

-- CPU版本(一行即可)
cmd -> pip install torch


-- 補充: 如果有裝錯,那就把環境移除重裝,ENVIRONMENT 自行修改為您的環境名稱
-- conda remove --name ENVIRONMENT --all


最後可以驗證一下是否有成功安裝好。

> python
> 
> import torch
> 
> print(torch.cuda.is_available())

如果要確認環境的實體位址

-- 確認目前環境的路徑
1. 開啟Anaconda prompt,輸入 where conda
2. conda activate XXX,再輸入 where conda

上一篇: [AI人工智能] 一、Python、Anaconda、Pycharm安裝

下一篇: [AI人工智能] 三、Aanconda, Pycharm 切換py環境


#Install #Python #PyTorch







Related Posts

從 Hoisting 理解 JavaScript 底層運作機制

從 Hoisting 理解 JavaScript 底層運作機制

Fake Vapes and How to Avoid Them

Fake Vapes and How to Avoid Them

LeetCode JS Easy 2704. To Be Or Not To Be

LeetCode JS Easy 2704. To Be Or Not To Be


Comments